from celery_tasks import constants
from celery_tasks.main import celery_app
from celery_tasks.sms.yuntongxun.sms import CCP
import logging

# 获取日志器对象
logger = logging.getLogger('django')


# 封装任务函数
@celery_app.task(name='sms_send_code')
def sms_send_code(mobile, sms_code, expire):
    print('任务函数调用了')
    # 发送短信
    try:
        # 调用 传入手机号, 发送验证码有效时间,使用的模板
        reg = CCP().send_template_sms(mobile, [sms_code, expire], constants.SEND_SMS_TEMP_ID)

    except BaseException as e:
        # 日志记录
        logger.error('短信发送失败 mobile:%s sms_code:%s ' % (mobile, sms_code))

    else:
        # 如果返回值不是0
        if reg != 0:
            # 日志记录发送失败
            logger.error('短信发送失败 mobile:%s sms_code:%s ' % (mobile, sms_code))
        # 否则
        else:
            # 记录发送成功
            logger.error('短信发送成功 mobile:%s sms_code:%s ' % (mobile, sms_code))

